from PIL import Image, ImageDraw, ImageFont

''' 给图片加入文字 '''


class ImgText:

    def __init__(self, text, note_img, text_width, font, text_color=None):
        # 预设文字宽度
        self.width = text_width
        # 文本
        self.text = text
        # 文本字体 例如：ImageFont.truetype("micross.ttf", 24)
        self.font = font
        # 文本颜色
        self.text_color = text_color
        # 段落 , 行数高, 行高, 行数
        self.duanluo, self.note_height, self.line_height, self.line_count, self.char_width = self.split_text()
        # 图片对象
        self.note_img = note_img

    def get_duanluo(self, text):
        txt = Image.new('RGBA', (100, 100))
        draw = ImageDraw.Draw(txt)
        # 所有文字的段落
        duanluo = ""
        # 宽度总和
        sum_width = 0
        # 几行
        line_count = 1
        # 行高
        line_height = 0
        for char in text:
            width, height = draw.textsize(char, self.font)
            sum_width += width
            if sum_width > self.width:  # 超过预设宽度就修改段落 以及当前行数
                line_count += 1
                sum_width = 0
                duanluo += '\n'
            duanluo += char
            line_height = max(height, line_height)
        if not duanluo.endswith('\n'):
            duanluo += '\n'
        return duanluo, line_height, line_count, sum_width

    def split_text(self):
        # 按规定宽度分组
        max_line_height, total_lines = 0, 0
        allText = []
        for text in self.text.split('\n'):
            duanluo, line_height, line_count, char_width = self.get_duanluo(
                text)
            max_line_height = max(line_height, max_line_height)
            total_lines += line_count
            allText.append((duanluo, line_count))
        line_height = max_line_height
        total_height = total_lines * line_height
        return allText, total_height, line_height, line_count, char_width

    # 获取总行高度
    def get_note_height(self):
        return self.note_height

    # 获取每行高度
    def get_line_height(self):
        return self.line_height

    # 获取行数
    def get_line_count(self):
        return self.line_count

    # 获取文字在每行中实际占的长度
    def get_char_width(self):
        return self.char_width

    # 在图片中画文字
    def draw_text(self, x, y):
        """
        绘图以及文字
        :return:
        """
        note_img = self.note_img
        draw = ImageDraw.Draw(note_img)
        # 根据传入的位置开始
        for duanluo, line_count in self.duanluo:
            draw.text((x, y), duanluo, fill=self.text_color, font=self.font)
            y += self.line_height * line_count
        # note_img.save("result.png")
        return note_img
